Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make stuff work more easily on Linux/ROCm #81

Open
wants to merge 2 commits into
base: main
Choose a base branch
from

Conversation

infinity0
Copy link

@infinity0 infinity0 commented Nov 27, 2024

With my relatively new AMD Radeon RX 7800 XT I need to set the following envvars for ROCm to work:

export PYTORCH_HIP_ALLOC_CONF=expandable_segments:True
export HSA_OVERRIDE_GFX_VERSION=11.0.0
export HCC_AMDGPU_TARGET=gfx1101

Instead of hardcoding this all, just pass the whole parent environment in, this is the default behaviour anyway for most "spawn" APIs.

Also torchaudio can't detect my ffmpeg 7, it seems to not support this yet. libsox-dev works as an alternative.

Now I can demux stems in 20 seconds compared to 10 minutes for Wine/CPU! :D

@infinity0 infinity0 mentioned this pull request Nov 27, 2024
@iffyloop
Copy link
Member

iffyloop commented Dec 2, 2024

Interesting... I suppose that was naive of me to not expect people to need to customize environment variables for their GPU. I feel like there was a reason I wanted to isolate the Demucs process from the rest of the system environment variables (which are kind of messy on Windows) but I'm not sure that my concern was actually valid. I'm hesitant to push out a new release and notify everyone to download 1-2GB update for this change only, but I'll merge this PR in on the next release.

Thank you so much for submitting this, and I'm happy to hear it's working now on your ROCm Linux device!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants